'''
视图模块：视图类主要与序列化器交互
'''

import random
from django.shortcuts import render

# Create your views here.
from django_redis import get_redis_connection # 连接redis要导入的包get_redis_connection('键名')来指定
from rest_framework import status
from rest_framework.decorators import action
from rest_framework.generics import CreateAPIView, RetrieveAPIView, UpdateAPIView  # 用户个人信息详情
from rest_framework.permissions import IsAuthenticated # 用户登录认证
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework.viewsets import ReadOnlyModelViewSet, ModelViewSet  # 视图集
# drf-extensions扩展对于缓存提供了三个扩展类：ListCacheResponseMixin，RetrieveCacheResponseMixin，CacheResponseMixin
from rest_framework_extensions.cache.mixins import CacheResponseMixin # 视图集扩展类，配合缓存list和retrieve，添加cache_response装饰器

from .serializers import CreateUserSerializer, UserDetailSerializer, EmailSerializer, \
    AreaSerializer, SubAreaSerializer, UserAddressSerializer, AddressTitleSerializer  # 指向哪个serializers.py的哪个序列化类
from .constants import SMS_CODE_REDIS_EXPIRES, SEND_SMS_CODE_INTERVAL, USER_ADDRESS_COUNTS_LIMIT
from .models import User, Area
# import logging
from celery_tasks.sms.yuntongxun.sms import CCP
from celery_tasks.sms.tasks import send_sms_code # 导入短信验证码发送的方法


# logger = logging.getLogger('django')

# 用户统计视图（判断用户是否存在）
class UsernameCountView(APIView):
    """
    用户名数量
    """
    def get(self, request, username):
        """
        获取指定用户名数量
        """
        count = User.objects.filter(username=username).count()

        data = {
            'username': username,
            'count': count
        }

        return Response(data)
# 返回数据：
#     {
#       "username": "itcast",
#       "count": "1"
#      }


# 手机号统计视图（判断手机号是否存在）
class MobileCountView(APIView):
    """
    手机号数量
    """
    def get(self, request, mobile):
        """
        获取指定手机号数量
        """
        count = User.objects.filter(mobile=mobile).count()

        data = {
            'mobile': mobile,
            'count': count
        }

        return Response(data)
# 返回数据：
#     {
#     "mobile": "18512345678",
#     "count": 0
#      }
class UserView(CreateAPIView):
    """
    用户注册
    传入参数：
        username, password, password2, sms_code, mobile, allow
    """
    serializer_class = CreateUserSerializer
# 返回数据：
# {
#     "id": 9,
#     "username": "python8",
#     "mobile": "18512345678",
# }

# 实现发送短信验证码的视图类
class SMSCodeView(APIView):
    """
    发送短信验证码
    1，生成和发送短信验证码：
        %06d表示输出的整数显示位数为6位，如果不足6位，则在左边用0补全；如果超过6位
    2，保存短信验证码：使用redis缓存，可设置失效时间
    3，redis pineline的使用：保存2个东西：记录和发送短信验证码
    4，检查是否在60s内有发送记录
    5，Celery异步发送短信
    状态码：用来识别请求是否成功：有可能被CORS策略阻止（2个不同的服务器之间的跨域访问）
    CORS:前端与后端分处不同的域名，我们需要为后端添加跨域访问的支持
        1，安装包：pip install django-cors-headers（安装完需要重新加载解析器）
        2，配置文件设置：
            1，添加子应用：'corsheaders'
            2，添加中间件：'corsheaders.middleware.CorsMiddleware',（第一行）
            3，添加白名单：前端服务器地址即可：'http://127.0.0.1:8081'（被CORS）
            4，设置参数支持cookie携带：CORS_ALLOW_CREDENTIALS = True
    """

    def get(self, request, mobile):

        # 创建连接到redis的对象（在配置文件中CACHES中）
        redis_conn = get_redis_connection('verify_codes')

        # # 60秒内不允许重发发送短信(60秒内刷新网页输入相同号码再点击就触发)
        send_flag = redis_conn.get('send_flag_%s' % mobile) # 进入redis中看这个键是否有值
        if send_flag:
            return Response({"message": "发送短信过于频繁"}, status=status.HTTP_400_BAD_REQUEST)

        # 生成和发送短信验证码（%06d表示输出的整数显示位数为6位，如果不足6位，则在左边用0补全；）
        sms_code = '%06d' % random.randint(0,999999)
        # 打印查看短信验证码
        print('短信验证码：',sms_code)
        # logger.debug(sms_code)
        # logging.info(sms_code)

        # 云通讯发送短信：3个参数--手机号，数组内容，模板id
        # 短信验证码有效期 300秒，短信是否发送过的标记60秒（这两个值可以另外外面py文件自定义字段修改也行）
        CCP().send_template_sms(mobile,[sms_code, SMS_CODE_REDIS_EXPIRES // SEND_SMS_CODE_INTERVAL], SMS_CODE_REDIS_EXPIRES)

        # 以下代码演示redis管道pipeline的使用（保存2个东西：记录和发送）
        pl = redis_conn.pipeline()
        # 保存短信验证码，并设置过期时间（过期时间也可在别的类来设置，也可固定）
        # 在redis中保存了一个键为"sms_18867455863" 值为验证码，300秒（5分钟）后过期
        pl.setex("sms_%s" % mobile, SMS_CODE_REDIS_EXPIRES, sms_code)
        # 保存记录的有效时间：60秒（也可在别的类来设置，也可固定）发送的记录保存的值设置为1
        # 在redis中保存了一个键为"send_flag_18867455863" 值为1，60秒（1分钟）后过期
        pl.setex('send_flag_%s' % mobile, SEND_SMS_CODE_INTERVAL, 1)
        # 执行以上redis的操作
        pl.execute()
        # 调用短信验证码发送的方法
        # send_sms_code.delay(mobile, sms_code)
        send_sms_code(mobile, sms_code)
        # 响应发送短信验证码结果
        return Response({"message": "OK"})


class UserDetailView(RetrieveAPIView):
    """
    用户详情
    """
    serializer_class = UserDetailSerializer
    # 添加用户权限（不然发送不了邮箱）
    permission_classes = [IsAuthenticated]

    def get_object(self):
        return self.request.user



class EmailView(UpdateAPIView):
    """
    保存用户邮箱功能：（根据当前登录用户的对应关系）
    保存邮箱数据
    返回数据
    """
    # 添加用户权限（不然发送不了邮箱）
    permission_classes = (IsAuthenticated,)

    serializer_class = EmailSerializer

    def get_object(self, *args, **kwargs):
        return self.request.user


class VerifyEmailView(APIView):
    """
    邮箱验证：用户在邮箱里点击链接，会回到用户个人信息，并在待验证显示为'已验证'
    """
    def get(self, request):

        # 获取token
        token = request.query_params.get('token')
        print(token)
        if not token:
            return Response({'message': '缺少token'}, status=status.HTTP_400_BAD_REQUEST)

        # 验证token（调用User模型类的验证token的方法）
        user = User().check_verify_email_token(token)
        if user is None:
            return Response({'message': '链接信息无效'}, status=status.HTTP_400_BAD_REQUEST)
        else:
            user.email_active = True
            user.save()
            return Response({'message': 'OK'})


# 行政区区域信息视图（需要继承使用drf-extensions提供的扩展类CacheResponseMixin，添加cache_response装饰器使用缓存，减少对数据库的访问）
class AreasViewSet(ReadOnlyModelViewSet,CacheResponseMixin):
    """
    行政区划信息（查询省市区的数据）
    查询省的语法和市区查询语法不同
    查询接口只有一个，且返回的数据有省而且有市区：用视图集实现，区分不同的请求方法：
    ReadOnlyModelViewSet--实现查询所有，单个，或一部分
    CacheResponseMixin--省市区的数据是经常被用户查询使用的，而且数据基本不变化，所以我们可以将省市区数据进行缓存处理，减少数据库的查询次数
    """
    pagination_class = None  # 区划信息不分页

    def get_queryset(self):
        """
        提供数据集：
        通过查询集的方法，返回：GET--查询所有--查询单个
        """
        # queryset = Area.objects.all()

        #  self.action 获取不同的请求方式：按需求返回数据
        if self.action == 'list':
            # 查询省
            return Area.objects.filter(parent=None)
        else:
            # 查询市/区
            return Area.objects.all()

    def get_serializer_class(self):
        """
        提供序列化器
        """
        if self.action == 'list':
            # 返回省的序列化器
            return AreaSerializer
        else:
            # 返回市区的序列化器
            return SubAreaSerializer


class AddressViewSet(ModelViewSet):
    """
    用户地址新增与修改
    GenericViewSet 使用GenericViewSet 视图集实现增删改查都在一个路由里面

    """
    serializer_class = UserAddressSerializer
    permissions = [IsAuthenticated]
    # queryset = Address.objects.all()
    # 查询登录用户的地址数据---
    # 查询方式： 使用当前登录的用户去查询与之关联的用户地址管理表
    # 改写 get_queryset 的返回值 让他只返回登录用户的地址数据
    def get_queryset(self):
        #
        return self.request.user.addresses.filter(is_deleted=False)

    # GET /addresses/
    def list(self, request, *args, **kwargs):
        """
        用户地址列表数据
        """
        queryset = self.get_queryset()
        serializer = self.get_serializer(queryset, many=True)
        # 获取登录用户的数据 user  是用户模型类对象
        user = self.request.user
        return Response({
            # 获取用户的id
            'user_id': user.id,
            # 获取
            'default_address_id': user.default_address_id,
            'limit': USER_ADDRESS_COUNTS_LIMIT,
            'addresses': serializer.data,
        })

    # POST /addresses/
    # 添加数据
    def create(self, request, *args, **kwargs):
        """
        保存用户地址数据
        """
        # 检查用户地址数据数目不能超过上限
        count = request.user.addresses.count()
        if count >= USER_ADDRESS_COUNTS_LIMIT:
            return Response({'message': '保存地址数据已达到上限'}, status=status.HTTP_400_BAD_REQUEST)
        # 返回添加后的数据 ,调用继承类中的create方法
        return super().create(request, *args, **kwargs)

    # delete /addresses/<pk>/
    def destroy(self, request, *args, **kwargs):
        """
        处理删除
        """
        # 获取 要删除的数据
        address = self.get_object()

        # 进行逻辑删除
        address.is_deleted = True
        address.save()

        return Response(status=status.HTTP_204_NO_CONTENT)

    # put /addresses/pk/status/
    @action(methods=['put'], detail=True)
    def status(self, request, pk=None):
        """
        修改设置默认地址
        """
        # 获取要设置为默认的数据
        address = self.get_object()
        # 为该用户添加的默认地址
        request.user.default_address = address
        # 保存
        request.user.save()
        return Response({'message': 'OK'}, status=status.HTTP_200_OK)

    # put /addresses/pk/title/
    # 需要请求体参数 title
    @action(methods=['put'], detail=True)
    def title(self, request, pk=None):
        """
        修改标题
        """
        address = self.get_object()
        serializer = AddressTitleSerializer(instance=address, data=request.data)
        serializer.is_valid(raise_exception=True)
        serializer.save()
        return Response(serializer.data)


# 用户修改密码视图
class PassWordView(APIView):
    """
    修改用户密码视图，请求接收：POST，ipassword，password，password2，
    效验原始密码ipassword正确否，
    正确的话效验新密码的正确性，返回ok，不正确返回error，
    """

    def put(self, request):
        # 获取参数
        ipassword = request.data["ipassword"]
        password = request.data["password"]
        password2 = request.data["password2"]
        user_id = request.data["user_id"]
        # 获得请求用户
        user = User.objects.get(id=user_id)
        # 效验新密码是否符合要求
        if password != password2:
            raise Exception("两次密码输入不一致！")
        if len(password) > 20 or len(password) < 8:
            raise Exception("密码长度需要8到20位")
        # 检查原始密码是否正确
        user.check_password(ipassword)

        # 修改密码为新密码
        user.set_password(password)
        user.save()

        # 返回数据
        return Response(status=status.HTTP_202_ACCEPTED)

# 最近浏览记录
class BrowseView(APIView):
    def get(self,request):
        return Response('记录一条浏览记录',status=status.HTTP_202_ACCEPTED)